#ifndef __SUBUNIT_H__
#define __SUBUNIT_H__

#include <string>

#include "unit.h"

namespace verema
{
/**
 * \class Subunit
 */
class Subunit : public Unit
{
private:
	int _unit_id;
	int _image_id;

public:
	Subunit()
	: Unit()
	, _unit_id(-1)
	, _image_id(-1)
	{ }

	Subunit(int id_, int unit_id_, int image_id_, std::string &name_, std::string &description_)
	: Unit(id_, name_, description_)
	, _unit_id(unit_id_)
	, _image_id(image_id_)
	{ }

	int unit_id() const { return _unit_id; }
	int image_id() const { return _image_id; }

	void unit_id(int unit_id_) { _unit_id = unit_id_; }
	void image_id(int image_id_) { _image_id = image_id_; }

	~Subunit()
	{ }
};

} // namespace verema

#endif // __SUBUNIT_H__
